import matplotlib.pyplot as plt
import numpy as np

def step_function(x):
    y = x > 0
    return y.astype(int)

x = np.arange(-5, 5, 0.1)
y = step_function(x)
plt.plot(x, y)
plt.ylim(-0.1, 1.1) # 指定 y 轴范围
plt.show()